#商品管理系统的开发
#1、商品添加   商品已存在，提示不能添加
#    名字：苹果  价格：5   库存：200
#2、查看商品   商品不存在？  按名字查看
#3、删除商品   按名字删除
#4、修改商品   按名字修改， 价格和库存
#5、退出
#可以不限次数输入
#pro_info = {"苹果":[5,200],"香蕉":[6,300]}
# pro_info = {"苹果":{"pro_price":5,"pro_stock":200}}

import pymysql
# 创建数据库连接
# host:连接的主机
conn = pymysql.connect(
    host="192.168.126.179",
    user = "root",
    password="123456",
    port = 3306,
    db = "product_db"
)



# 每个功能都被封装成了一个独立的函数，使得代码更加模块化和易于维护。
#1.添加商品的函数封装

def add_product(pro_name,pro_price,pro_stock):
    if pro_name in pro_info:
        return  False,"商品已存在"
    pro_info[pro_name] = {"pro_price":pro_price,"pro_stock":pro_stock }
    return True,"商品添加成功"

#2.查看商品的函数封装
def view_product(pro_name):
    if pro_name in pro_info:
        return True, f"商品 {pro_name} 的信息：价格 {pro_info[pro_name]['pro_price']}，库存 {pro_info[pro_name]['pro_stock']}"
    else:
        return False, "商品不存在！"

# 3.删除商品的函数封装
def delete_product(pro_name):
    if pro_name in pro_info:
        del pro_info[pro_name]
        return True, f"商品 {pro_name} 已被删除！"
    else:
        return False, "商品不存在！"


#4.修改商品的函数封装
def modify_product(pro_name, pro_price, pro_stock):
    if pro_name in pro_info:
        pro_info[pro_name]["pro_price"] = pro_price
        pro_info[pro_name]["pro_stock"] = pro_stock
        return True, f"商品 {pro_name} 的信息已更新！"
    else:
        return False, "商品不存在！"

pro_info = {}
while 1:
    choice = input("请输入你的选择：")
    # if choice == "5":
    #     break
        #1.商品添加
    if choice == "1":
        pro_name = input("请输入商品的名字：")
        pro_price = input("请输入商品的价格：")
        pro_stock = input("请输入商品的库存：")
        status,message = add_product(pro_name,pro_price,pro_stock)
        # pro_info[pro_name] = [pro_price,pro_stock]

        #检查商品是否已经存在
        if pro_name in pro_info:
            print("商品已经存在，不能重复添加！")
            continue
        #添加商品到字典
        pro_info[pro_name] = {"pro_price":pro_price,"pro_stock":pro_stock}
        if status:
            print(message)
        else:
            print(message)
        # 2. 查看商品
    elif choice == "2":
        pro_name = input("请输入要查看的商品名字: ")
        status, message = view_product(pro_name)
        if status:
            print(message)
        else:
            print(message)

    # 3. 删除商品
    elif choice == "3":
        pro_name = input("请输入要删除的商品名字: ")
        status, message = delete_product(pro_name)
        if status:
            print(message)
        else:
            print(message)

    # 4. 修改商品
    elif choice == "4":
        pro_name = input("请输入要修改的商品名字: ")
        if pro_name in pro_info:
            pro_price = input("请输入新的价格: ")
            pro_stock = input("请输入新的库存: ")
            status, message = modify_product(pro_name, pro_price, pro_stock)
            if status:
                print(message)
            else:
                print(message)
    # 5. 退出
    elif choice == "5":
        break
    else:
        print("无效的选择，请重新输入！")





